package cn.fastoo.web.backstage.web.controller.excel;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

/**
 *
 * @author hewangtong
 *
 */
public class ReadExcel {
    // 总行数
    private int totalRows = 0;
    // 总条数
    private int totalCells = 0;
    // 错误信息接收器
    private String errorMsg;

    // 构造方法
    public ReadExcel() {
    }

    // 获取总行数
    public int getTotalRows() {
        return totalRows;
    }

    // 获取总列数
    public int getTotalCells() {
        return totalCells;
    }

    // 获取错误信息
    public String getErrorInfo() {
        return errorMsg;
    }

    /**
     * 读EXCEL文件，获取信息集合
     *
     * @return
     */
    public List<Map<String, Object>> getExcelInfo(MultipartFile mFile) {
        String fileName = mFile.getOriginalFilename();// 获取文件名
//        List<Map<String, Object>> userList = new LinkedList<Map<String, Object>>();
        try {
            if (!validateExcel(fileName)) {// 验证文件名是否合格
                return null;
            }
            boolean isExcel2003 = true;// 根据文件名判断文件是2003版本还是2007版本
            if (isExcel2007(fileName)) {
                isExcel2003 = false;
            }
            return createExcel(mFile.getInputStream(), isExcel2003);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 根据excel里面的内容读取客户信息
     *
     * @param is      输入流
     * @param isExcel2003   excel是2003还是2007版本
     * @return
     * @throws IOException
     */
    public List<Map<String, Object>> createExcel(InputStream is, boolean isExcel2003) {
        try {
            Workbook wb = null;
            if (isExcel2003) {// 当excel是2003时,创建excel2003
                wb = new HSSFWorkbook(is);
            } else {// 当excel是2007时,创建excel2007
                wb = new XSSFWorkbook(is);
            }
            return readExcelValue(wb);// 读取Excel里面客户的信息
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 读取Excel里面客户的信息
     *
     * @param wb
     * @return
     */
    private List<Map<String, Object>> readExcelValue(Workbook wb) {
        // 得到第一个shell
        Sheet sheet = wb.getSheetAt(0);
        // 得到Excel的行数
        this.totalRows = sheet.getPhysicalNumberOfRows();
        // 得到Excel的列数(前提是有行数)
        if (totalRows > 1 && sheet.getRow(0) != null) {
            this.totalCells = sheet.getRow(0).getPhysicalNumberOfCells();
        }
        List<Map<String, Object>> userList = new ArrayList<Map<String, Object>>();
        // 循环Excel行数
        for (int r = 1; r < totalRows; r++) {
            Row row = sheet.getRow(r);
            if (row == null) {
                continue;
            }
            // 循环Excel的列
            Map<String, Object> map = new HashMap<String, Object>();
            for (int c = 0; c < this.totalCells; c++) {
                Cell cell = row.getCell(c);
                if (null != cell) {
                    if (c == 0) {
                        // 如果是纯数字,比如你写的是25,cell.getNumericCellValue()获得是25.0,通过截取字符串去掉.0获得25
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String username = String.valueOf(cell.getNumericCellValue());
                            map.put("username", username.substring(0, username.length() - 2 > 0 ? username.length() - 2 : 1));// 名称
                        } else {
                            map.put("username", cell.getStringCellValue());// 名称
                        }
                    } else if (c == 1) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String englishName = String.valueOf(cell.getNumericCellValue());
                            map.put("englishName",englishName.substring(0, englishName.length() - 2 > 0 ? englishName.length() - 2 : 1));// 英文名
                        } else {
                            map.put("englishName",cell.getStringCellValue());// 性别
                        }
                    } else if (c == 2) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String sex = String.valueOf(cell.getNumericCellValue());
                            map.put("sex", sex.substring(0, sex.length() - 2 > 0 ? sex.length() - 2 : 1));// 性别
                        } else {
                            map.put("sex", cell.getStringCellValue());// 性别
                        }
                    } else if (c == 3) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String birthday = String.valueOf(cell.getNumericCellValue());
                            map.put("birthday", birthday.substring(0, birthday.length() - 2 > 0 ? birthday.length() - 2 : 1));// 生日
                        } else {
                            map.put("birthday", cell.getStringCellValue());// 生日
                        }
                    }else if (c == 4) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String nationality = String.valueOf(cell.getNumericCellValue());
                            map.put("nationality", nationality.substring(0, nationality.length() - 2 > 0 ? nationality.length() - 2 : 1));// 国籍
                        } else {
                            map.put("nationality", cell.getStringCellValue());// 国籍
                        }
                    }else if (c == 5) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String isSchool = String.valueOf(cell.getNumericCellValue());
                            map.put("isSchool", isSchool.substring(0, isSchool.length() - 2 > 0 ? isSchool.length() - 2 : 1));// 学校
                        } else {
                            map.put("isSchool", cell.getStringCellValue());// 学校
                        }
                    }else if (c == 6) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String idCard = String.valueOf(cell.getNumericCellValue());
                            map.put("idCard", idCard.substring(0, idCard.length() - 2 > 0 ? idCard.length() - 2 : 1));// 身份证
                        } else {
                            map.put("idCard", cell.getStringCellValue());
                        }
                    }else if (c == 7) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String parent = String.valueOf(cell.getNumericCellValue());
                            map.put("parent", parent.substring(0, parent.length() - 2 > 0 ? parent.length() - 2 : 1));// 监护人
                        } else {
                            map.put("parent", cell.getStringCellValue());
                        }
                    }else if (c == 8) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String handset = String.valueOf(cell.getNumericCellValue());
                            map.put("handset", handset.substring(0, handset.length() - 2 > 0 ? handset.length() - 2 : 1));// 电话号码
                        } else {
                            map.put("handset", cell.getStringCellValue());
                        }
                    }else if (c == 9) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String parentTwo = String.valueOf(cell.getNumericCellValue());
                            map.put("parentTwo", parentTwo.substring(0, parentTwo.length() - 2 > 0 ? parentTwo.length() - 2 : 1));//  监护人2
                        } else {
                            map.put("parentTwo", cell.getStringCellValue());
                        }
                    }else if (c == 10) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String handsetTwo = String.valueOf(cell.getNumericCellValue());
                            map.put("handsetTwo", handsetTwo.substring(0, handsetTwo.length() - 2 > 0 ? handsetTwo.length() - 2 : 1));// 电话号码2
                        } else {
                            map.put("handsetTwo", cell.getStringCellValue());
                        }
                    }else if (c == 11) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String wechat = String.valueOf(cell.getNumericCellValue());
                            map.put("wechat", wechat.substring(0, wechat.length() - 2 > 0 ? wechat.length() - 2 : 1));// 微信
                        } else {
                            map.put("wechat", cell.getStringCellValue());
                        }
                    }else if (c == 12) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String address = String.valueOf(cell.getNumericCellValue());
                            map.put("address", address.substring(0, address.length() - 2 > 0 ? address.length() - 2 : 1));// 地址
                        } else {
                            map.put("address", cell.getStringCellValue());
                        }
                    }else if (c == 13) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String campus = String.valueOf(cell.getNumericCellValue());
                            map.put("campus", campus.substring(0, campus.length() - 2 > 0 ? campus.length() - 2 : 1));// 校区
                        } else {
                            map.put("campus", cell.getStringCellValue());
                        }
                    }else if (c == 14) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String collectTime = String.valueOf(cell.getNumericCellValue());
                            map.put("collectTime", collectTime.substring(0, collectTime.length() - 2 > 0 ? collectTime.length() - 2 : 1));// 收集时间
                        } else {
                            map.put("collectTime", cell.getStringCellValue());
                        }
                    } else if (c == 15) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String introducer = String.valueOf(cell.getNumericCellValue());
                            map.put("introducer", introducer.substring(0, introducer.length() - 2 > 0 ? introducer.length() - 2 : 1));// 转介绍人
                        } else {
                            map.put("introducer", cell.getStringCellValue());
                        }
                    }else if (c == 16) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String activityLocation = String.valueOf(cell.getNumericCellValue());
                            map.put("activityLocation", activityLocation.substring(0, activityLocation.length() - 2 > 0 ? activityLocation.length() - 2 : 1));// 活动位置
                        } else {
                            map.put("activityLocation", cell.getStringCellValue());
                        }
                    }else if (c == 17) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String collectPeople = String.valueOf(cell.getNumericCellValue());
                            map.put("collectPeople", collectPeople.substring(0, collectPeople.length() - 2 > 0 ? collectPeople.length() - 2 : 1));// 收集人员
                        } else {
                            map.put("collectPeople", cell.getStringCellValue());
                        }
                    }else if (c == 18) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String consultantGetMan = String.valueOf(cell.getNumericCellValue());
                            map.put("consultantGetMan", consultantGetMan.substring(0, consultantGetMan.length() - 2 > 0 ? consultantGetMan.length() - 2 : 1));// 顾问获取人
                        } else {
                            map.put("consultantGetMan", cell.getStringCellValue());
                        }
                    }else if (c == 19) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String originDate = String.valueOf(cell.getNumericCellValue());
                            map.put("originDate", originDate.substring(0, originDate.length() - 2 > 0 ? originDate.length() - 2 : 1));// 来源日期
                        } else {
                            map.put("originDate", cell.getStringCellValue());
                        }
                    }else if (c == 20) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String originNotes = String.valueOf(cell.getNumericCellValue());
                            map.put("originNotes", originNotes.substring(0, originNotes.length() - 2 > 0 ? originNotes.length() - 2 : 1));// 来源情况记录
                        } else {
                            map.put("originNotes", cell.getStringCellValue());
                        }
                    }else if (c == 21) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String remark = String.valueOf(cell.getNumericCellValue());
                            map.put("remark", remark.substring(0, remark.length() - 2 > 0 ? remark.length() - 2 : 1));// 备注
                        } else {
                            map.put("remark", cell.getStringCellValue());
                        }
                    }else if (c == 22) {
                        if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                            String status = String.valueOf(cell.getNumericCellValue());
                            map.put("status", status.substring(0, status.length() - 2 > 0 ? status.length() - 2 : 1));// 客户状态
                        } else {
                            map.put("status", cell.getStringCellValue());
                        }
                    }


                }
            }
            // 添加到list
            userList.add(map);
        }
        return userList;
    }

    /**
     * 验证EXCEL文件
     *
     * @param filePath
     * @return
     */
    public boolean validateExcel(String filePath) {
        if (filePath == null || !(isExcel2003(filePath) || isExcel2007(filePath))) {
            errorMsg = "文件名不是excel格式";
            return false;
        }
        return true;
    }

    // @描述：是否是2003的excel，返回true是2003
    public static boolean isExcel2003(String filePath) {
        return filePath.matches("^.+\\.(?i)(xls)$");
    }

    // @描述：是否是2007的excel，返回true是2007
    public static boolean isExcel2007(String filePath) {
        return filePath.matches("^.+\\.(?i)(xlsx)$");
    }

}
